python use variables for basic math